package com.it.zhangjun.arithmetic.回溯;

/**
 * 八皇后
 *
 * @author jun.zhang6
 * @date 2020/10/27
 */
public class Queues {
    public static void main(String[] args) {
        Queues queues = new Queues();
        queues.placeQueues(4);
    }

    /**
     * 存放每一个黄后的列号，在第几列
     */
    private int[] queues;
    /**
     * 一共有多少种合理的摆法
     */
    private int ways;

    /**
     * 摆放皇后
     */
    private void placeQueues(int count) {
        if (count < 1) {
            return;
        }
        queues = new int[count];

        place(0);
        System.out.println(count + "皇后一共有" + ways + "种摆法");
    }

    /**
     * 从第row行开始摆放皇后
     */
    private void place(int row) {
        if (row == queues.length) {
            ways++;
            show();
            return;
        }
        for (int col = 0; col < queues.length; col++) {
            if (isValid(row, col)) {
                queues[row] = col;
                place(row + 1);
            }
        }
    }

    /**
     * 检查第row行第col列是否可以摆放皇后
     */
    private boolean isValid(int row, int col) {
        for (int i = 0; i < row; i++) {
            if (queues[i] == col) {
                return false;
            }
            if (row - i == Math.abs(col - queues[i])) {
                return false;
            }
        }
        return true;
    }

    /**
     * 显示皇后的摆放情况
     */
    private void show() {
        for (int row = 0; row < queues.length; row++) {
            for (int col = 0; col < queues.length; col++) {
                if (queues[row] == col) {
                    System.out.print(1);
                } else {
                    System.out.print(0);
                }
                System.out.print(" ");
            }
            System.out.println();
        }
        System.out.println("-----------------------------------");
    }
}
